diff --git a/hikey960.mk b/hikey960.mk
index 3a58d25..673181c 100644
--- a/hikey960.mk
+++ b/hikey960.mk
@@ -8,6 +8,10 @@
 ifndef HIKEY_USES_GKI
   ifeq ($(TARGET_KERNEL_USE), 5.4)
     HIKEY_USES_GKI := true
+  else
+    ifeq ($(TARGET_KERNEL_USE), mainline)
+      HIKEY_USES_GKI := true
+    endif
   endif
 endif